package com.qfedu.repo.db;

import java.sql.*;
import java.util.List;

/**
 * 该类为攻击类，作用是：通过手敲代码实现jdbc的数据库连接
 */
public class DBUtils {
    private static final String MYSQL_DRIVER = "com.mysql.cj.jdbc.Driver";
    private static final String MYSQL_URL = "jdbc:mysql://127.0.0.1:3306/repo";
    private static final String MYSQL_USER = "root";
    private static final String MYSQL_PWD = "123456";
    private static Connection conn = null;
    private static PreparedStatement pst = null;


    /**
     * jdbc连接数据库的步骤：
     * 1、加载驱动
     * 2、简历数据库的连接：url、user、pwd，返回数据库连接对象
     * 3、获取数据库预编译执行pst对象
     * 4、执行数据库操作int update(sql)、ResultSet query(sql)
     * 5、根据结果处理
     * 6、关闭连接、释放对象
     */
    static {
        try {
            Class.forName(MYSQL_DRIVER);
        } catch (ClassNotFoundException e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 该方法用于建立与数据库的连接，并返回连接对象
     *
     * @return 建立的数据库连接对象
     */
    public static Connection getConnection() {
        Connection connection = null;
        try {
            connection = DriverManager.getConnection(MYSQL_URL, MYSQL_USER, MYSQL_PWD);
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }
        return connection;
    }

    /**
     * 该方法用于执行数据库操作，接收两个参数，第一个参数带有? 的sql语句
     *
     * @param sql      具体要执行的sql语句
     * @param callback 回调对象，用于调用者动态灵活的设置? 的数据
     * @return
     */
    public static int executeUpdate(String sql, PstCallback callback) {
        conn = getConnection();
        int result = -1;
        if (conn != null) {
            try {
                //预编译执行sql语句
                pst = conn.prepareStatement(sql);
                //调用一系列的pst的.set方法给?占位符赋值
                if (conn != null) {
                    callback.pstSetValue(pst);
                }
                result=pst.executeUpdate();
            } catch (SQLException e) {
                throw new RuntimeException(e);
            }finally {//在finally代码块中，关闭数据库连接
                close();
            }
        }
        return result;
    }

    /**
     * 该方法用于执行数据库的次查询，会将数据库的查询结果统一返回
     * @param sql 要执行的sql语句
     * @param pstCallback 设置sql中的变量的回调
     * @return
     */
    public static ResultSet query(String sql, PstCallback pstCallback){
        ResultSet result=null;
        try {
            conn=getConnection();
            pst=conn.prepareStatement(sql);
            //将pst对象回传给调用者，供调用者设置?对应的数据
            if (pstCallback != null) {
                pstCallback.pstSetValue(pst);
            }

            result=pst.executeQuery();
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }
        return result;
    }

    /**
     * 该方法用于关闭数据库连接
     */
    public static void close(){
        try {
            if(pst!=null){
                pst.close();
            }
            if(conn!=null){
                conn.close();
            }
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }
    }

    public static void main(String[] args) {
        System.out.println(getConnection());
    }
}
